import pymysql
from API.common.config import url_config

class DoMysql:
    # 这里可以通过配置文件或者传参的方式来封装，但是我们用配置文件比较好管理
    def __init__(self):

        host = url_config.get('mysql','host')  # 数据库的ip地址
        user = url_config.get('mysql','user') # 数据库的账号
        password = url_config.get('mysql','password')  # 数据库的密码
        port = url_config.int('mysql','port')  # mysql数据库的端口号
        self.mysql = pymysql.connect (host=host, user=user, password=password, port=port)
        self.cursor = self.mysql.cursor ()

    # 返回单条数据
    def fetch_one(self, sql):
        self.cursor.execute (sql)
        return self.cursor.fetchone ()

    # 返回多条数据
    def fetch_chall(self, sql):
        self.cursor.execute (sql)
        return self.cursor.fetchall ()

    def fetch_code(self):
        self.cursor.close ()
        self.mysql.close ()